package com.tianji.aigc.service.impl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.IdUtil;

import com.tianji.aigc.entity.ChatSession;
import com.tianji.aigc.mapper.ChatSessionMapper;

import com.tianji.aigc.memory.RedisChatMemory;
import com.tianji.aigc.properties.SessionProperties;
import com.tianji.aigc.service.IChatSessionService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import com.tianji.aigc.vo.MessageVO;
import com.tianji.aigc.vo.SessionVO;
import com.tianji.common.exceptions.BadRequestException;
import com.tianji.common.utils.UserContext;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.MessageType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Collections;
import java.util.List;

/**
 * <p>
 * 对话session 服务实现类
 * </p>
 *
 * @author mr.wu
 * @since 2025-09-02
 */
@Service
public class ChatSessionServiceImpl extends ServiceImpl<ChatSessionMapper, ChatSession> implements IChatSessionService {

    @Autowired
    private SessionProperties sessionProperties;

    /**
     * 新建会话
     *
     * @param num
     * @return
     */
    @Override
    public SessionVO newSession(Integer num) {
        //1.生成唯一会话ID
        String sessionId = IdUtil.fastSimpleUUID(); //唯一的会话ID，使用UUID算法

        //2.新增会话记录
        ChatSession chatSession = ChatSession.builder()
                .userId(UserContext.getUser())//用户id
                .sessionId(sessionId)//会话id
                .build();
        this.save(chatSession);

        //3.随机获取一批热门问题
        List<SessionVO.Example> examples = sessionProperties.getExamples();
        Collections.shuffle(examples);//打乱集合中元素的顺序（随机性）
        List<SessionVO.Example> exampleList = examples.stream().limit(num).toList(); //前端需要的随机N条热门问题列表

        //4.封装响应结果数据
        SessionVO sessionVO = SessionVO.builder()
                .sessionId(sessionId)//会话id
                .title(sessionProperties.getTitle())//默认标题
                .describe(sessionProperties.getDescribe())//默认描述
                .examples(exampleList)//随机问题
                .build();

        return sessionVO;
    }


    /**
     * 热门问题
     *
     * @param num
     * @return
     */
    @Override
    public List<SessionVO.Example> hotExampleList(Integer num) {
        List<SessionVO.Example> examples = sessionProperties.getExamples();
        Collections.shuffle(examples);//打乱集合中元素的顺序（随机性）
        List<SessionVO.Example> exampleList = examples.stream().limit(num).toList(); //前端需要的随机N条热门问题列表
        return exampleList;
    }
    @Autowired
    private RedisChatMemory redisChatMemory;
    /**
     * 从redis中查询会话记录
     *
     * @param sessionId ;
     */
    @Override
    public List<MessageVO> getMessageList(String sessionId) {
        //获取对话ID(由用户id加上会话id组成)
        String conversationId = UserContext.getUser() + ":" + sessionId;
        //判断对话id是否是kong
       if (conversationId==null){
           throw new BadRequestException("暂未有对话记录");
       }
       //根据对话id查询记录
        List<Message> messageList = redisChatMemory.get(conversationId, 100);
       //消息转化
        List<MessageVO> messageVOList = messageList.stream()
                .filter(x -> x.getMessageType().equals(MessageType.USER) || x.getMessageType().equals(MessageType.ASSISTANT))
                .map(x -> MessageVO.builder()
                        .type(x.getMessageType())
                        .content(x.getText())
                        .build())
                .toList();

        return messageVOList;
    }
}
